package org.example.utils;

import java.util.UUID;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component
public class TokenUtil {
    @Autowired
    private RedisUtil redisUtil;
    /**
     * 存入 Redis 的 Token 键的前缀
     */
    private static final String TOKEN_PREFIX = "idempotent_token::";
    private static final String MSG_CODE_PREFIX = "msg_code::";

    /**
     * 创建 Token 存入 Redis，并返回该 Token
     */
    public String generateToken(Object value, Long expireTime) {
        // 实例化生成 ID 工具对象
        String token = UUID.randomUUID().toString();
        // 设置存入 Redis 的 Key
        String key = TOKEN_PREFIX + token;
        // 存储 Token 到 Redis，且设置过期时间为30分钟
        redisUtil.set(key, value, expireTime);
        // 返回 Token
        return token;
    }

    public String generateMsgToken(Object value, Long expireTime) {
        // 实例化生成 ID 工具对象
        String token = UUID.randomUUID().toString();
        // 设置存入 Redis 的 Key
        String key = MSG_CODE_PREFIX + token;
        // 存储 Token 到 Redis，且设置过期时间为30分钟
        redisUtil.set(key, value, expireTime);
        // 返回 Token
        return token;
    }

    /**
     * 验证 Token 正确性
     */
    public boolean validToken(String token, Object value) {
        String truthVal = (String) redisUtil.get(TOKEN_PREFIX + token);
        if (truthVal == null || "".equals(truthVal)) {
            return false;
        }
        if (!value.equals(truthVal)) {
            return false;
        }
        return true;
    }

    public boolean validMsgToken(String token, Object value) {
        String truthVal = (String) redisUtil.get(MSG_CODE_PREFIX + token);
        if (truthVal == null || "".equals(truthVal)) {
            return false;
        }
        if (!value.equals(truthVal)) {
            return false;
        }
        return true;
    }

    public void clearToken(String k) {
        redisUtil.remove(k);
    }

}